package com.czk.algorithm.kmp;

/**
 * KMP算法
 *
 * @Author:ChenZhangKun
 * @Date: 2020/7/30 21:57
 */
public class KMP {
    public static void main(String[] args) {
        String str1 = "BBC ABCDAB ABCDABCDABDE";
        String str2 = "ABCDABD";
        int[] ints = kmpNext(str2);
        int kmp = kmp(str1, str2, ints);
        System.out.println(kmp);
    }

    /**
     * @param str1 源串
     * @param st2  子串
     * @param next 部分匹配表
     * @return
     */
    public static int kmp(String str1, String st2, int[] next) {
        int length1 = str1.length();
        // i str1指针
        // j str2的指针
        for (int i = 0, j = 0; i < length1; i++) {
            // KMP算法的核心点，两者不相等的情况
            while (j > 0 && str1.charAt(i) != st2.charAt(j)) {
                j = next[j - 1];
            }
            if (str1.charAt(i) == st2.charAt(j)) {
                j++;
            }
            // 匹配完成
            if (j == st2.length()) {
                // 返回字符串的索引
                return i - j + 1;
            }
        }
        // 没有匹配到情况
        return -1;
    }

    /**
     * 部分匹配表的获取
     *
     * @param str2
     * @return
     */
    public static int[] kmpNext(String str2) {
        // 创建next保存部分匹配值
        int[] next = new int[str2.length()];
        // 如果字符串长度为1，他的部分匹配值为0
        next[0] = 0;
        for (int i = 1, j = 0; i < str2.length(); i++) {
            while (j > 0 && str2.charAt(i) != str2.charAt(j)) {
                j = next[j - 1];
            }
            // 相等时，部分匹配值加1
            if (str2.charAt(i) == str2.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }
}
